/*
 * 版权所有(C)，上海海鼎信息工程股份有限公司，lizhixing，所有权利保留。
 * <p>
 * 项目名：	boot-demo
 * 文件名：	DruidConfiguration
 * 模块说明：
 * 修改历史：
 * 2022/1/5 - lizhixing - 创建。
 */
package com.lovely602.boot.bootdemo.config;

import org.springframework.boot.web.servlet.FilterRegistrationBean;
import org.springframework.boot.web.servlet.ServletRegistrationBean;
import org.springframework.context.annotation.Bean;
import org.springframework.context.annotation.Configuration;

import com.alibaba.druid.support.http.StatViewServlet;
import com.alibaba.druid.support.http.WebStatFilter;

/**
 * @author lizhixing
 */
@Configuration
public class DruidConfiguration {

  @Bean
  public ServletRegistrationBean<StatViewServlet> druidStatViewServlet() {
    ServletRegistrationBean<StatViewServlet> servletRegistrationBean = new ServletRegistrationBean<>(
        new StatViewServlet(), "/druid/*");
    // 添加初始化参数
    // 白名单
    servletRegistrationBean.addInitParameter("allow", "127.0.0.1");
    servletRegistrationBean.addInitParameter("deny", "192.168.0.103");
    // 登录查看信息的账号密码
    servletRegistrationBean.addInitParameter("loginUserName", "admin");
    servletRegistrationBean.addInitParameter("loginPassword", "123456");

    // 是否能够重置数据
    servletRegistrationBean.addInitParameter("resetEnable", "false");
    return servletRegistrationBean;
  }

  @Bean
  public FilterRegistrationBean<WebStatFilter> druidStatFilter() {
    FilterRegistrationBean<WebStatFilter> filterRegistrationBean = new FilterRegistrationBean<>(
        new WebStatFilter());
    // 添加过滤规则
    filterRegistrationBean.addUrlPatterns("/*");

    // 添加需要忽略的格式信息
    filterRegistrationBean.addInitParameter("exclusions",
        "*.js,*.gif,*.jpg,*.png,*.css,*.ico,/druid/*");
    return filterRegistrationBean;
  }

}
